import numpy as np


def AtomCenter_Pos(Cam):
    if Cam == '22860':
        Atom_Center = []
        Atom_Center.append(np.array([274, 472, 0, 0]))  # 0, big
        Atom_Center.append(np.array([208, 474, 0, 0]))  # 1
        Atom_Center.append(np.array([273, 408, 0, 0]))  # 2
        Atom_Center.append(np.array([337, 470, 0, 0]))  # 3
        Atom_Center.append(np.array([273, 537, 0, 0]))  # 4
        Atom_Center.append(np.array([354, 302, 0, 0]))  # 5
        Atom_Center.append(np.array([421, 303, 0, 0]))  # 6
        Atom_Center.append(np.array([354, 371, 0, 0]))  # 7
        Atom_Center.append(np.array([422, 370, 0, 0]))  # 8

        Atom_Center.append(np.array([994, 563, 0, 0]))  # Background
    elif Cam == '22834':
        Atom_Center = []
        Atom_Center.append(np.array([665, 541, 0, 0]))  # 0, big
        Atom_Center.append(np.array([614, 539, 0, 0]))  # 1
        Atom_Center.append(np.array([668, 439, 0, 0]))  # 2
        Atom_Center.append(np.array([716, 541, 0, 0]))  # 3
        Atom_Center.append(np.array([664, 641, 0, 0]))  # 4
        Atom_Center.append(np.array([423, 443, 0, 0]))  # 5
        Atom_Center.append(np.array([476, 447, 0, 0]))  # 6
        Atom_Center.append(np.array([529, 446, 0, 0]))  # 7
        Atom_Center.append(np.array([422, 546, 0, 0]))  # 8
        Atom_Center.append(np.array([475, 547, 0, 0]))  # 9
        Atom_Center.append(np.array([526, 549, 0, 0]))  # 10

        Atom_Center.append(np.array([994, 563, 0, 0]))  # Background
    return Atom_Center


def CutWidths(Cam):
    if Cam == '22860':
        Cut_Width = np.zeros(10)
        Cut_Width[0] = 45
        Cut_Width[1:5] = 35
        Cut_Width[5:9] = 30
        Cut_Width[9] = 50  # bg
        Cut_Width = Cut_Width.astype(int)
    elif Cam == '22834':
        Cut_Width = np.zeros(12)
        Cut_Width[0] = 50
        Cut_Width[1] = 32
        Cut_Width[2] = 30
        Cut_Width[3] = 30
        Cut_Width[4] = 32
        Cut_Width[5] = 20
        Cut_Width[6] = 24
        Cut_Width[7] = 18
        Cut_Width[8] = 17
        Cut_Width[9] = 20
        Cut_Width[10] = 16

        Cut_Width[11] = 50  # bg
        Cut_Width = Cut_Width.astype(int)
    return Cut_Width
